# -*- coding: utf-8  -*-
# Código GPL, autor: BigSus
# http://es.wikipedia.org/wiki/Usuario:Bigsus-bot/Variables.py

import wikipedia, time, re
from urllib import urlopen
from operator import itemgetter

# modo = None actualizará los listados de todos los wikis
modo = None

parametros = {
        'ca': {
                'pagina':    u'Plantilla:Variables/dades',
                'resumen':   u'Bot: Actualitzant plantilla',
                'principio': u'<!-- Para añadir un nuevo idioma, pídelo en la discusión, no lo añadas directamente. -->',
                'final':     u''
        },
        'en': {
                'pagina':    u'Template:NUMBEROF/data',
                'resumen':   u'Bot: Automatic update of variables',
                'principio': u'<!-- If you want to add a new language, ask for it in the talk pages, not here directly. -->',
                'final':     u''
        },
        'es': {
                'pagina':    u'Plantilla:Variables/datos',
                'resumen':   u'Bot: Actualización automática de variables',
                'principio': u'<!-- Para añadir un nuevo idioma, pídelo en la discusión, no lo añadas directamente. -->',
                'final':     u'\n\n[[Categoría:Wikipedia:Plantillas de apoyo|{{PAGENAME}}]]'
        },
        'tr': {
                'pagina':    u'Şablon:Viki istatistikleri/veri',
                'resumen':   u'Bot: Automatic update of variables',
                'principio': u'<!-- If you want to add a new language, ask for it in the talk pages, not here directly. -->',
                'final':     u'\n\n[[Kategori:Vikipedi şablonları]]'
        }
}

       
def generar_datos():
        """
        Recorrer todos los wikis en busca de los datos que necesitamos.
        Se devuelve un array con todos los datos para procesarlos después.
        """
        modo2 = modo
        if modo2==None:
                modo2 = 'todos'

        # Idiomas para revisar (listado automática)
        lista_idiomas = wikipedia.Site.languages(wikipedia.Site('es', 'wikipedia'))
        
        wikipedia.output('Iniciando variables.py...\nFecha y hora al comenzar el análisis: \03{lightred}%s\03{default}. Idioma: \03{lightred}%s\03{default}' % (time.strftime("%d-%m-%Y %H:%M:%S", time.localtime()), modo2))
        wikipedia.output('Se han encontrado %s idiomas para revisar.\n' % len(lista_idiomas))
        wikipedia.output('\t+---------+------------+-----------+\n\t|idioma ··|·· total  ··|·· arts. ··|\n\t+---------+------------+-----------+')

        # Variables comunes
        parserData = r"total=(?P<total>.*);good=(?P<good>.*);views=(?P<views>.*);edits=(?P<edits>.*);users=(?P<users>.*);activeusers=(?P<activeusers>.*);admins=(?P<admins>.*);images=(?P<images>.*);jobs=(?P<jobs>.*)"

        partesPage = r""

        articulosT = 0
        ficherosT = 0
        paginasT = 0
        usuariosT = 0
        usuariosactivosT = 0
        adminsT = 0
        editsT = 0

        datos = []
        cont = 0

        for i in lista_idiomas:
                texto = urlopen('http://%s.wikipedia.org/wiki/Special:Statistics?action=raw' % (i)).read()
                param = re.search(parserData, texto)
                wikipedia.output(' %s	\03{lightpurple}%s\03{default}	... %s	... %s' % (cont+1, i, param.group('total'), param.group('good')))

                datos.append([i, int(param.group('good')), int(param.group('images')), int(param.group('total')), int(param.group('users')), int(param.group('activeusers')), int(param.group('admins')), int(param.group('edits'))])

                # Calcular totales
                articulosT += datos[cont][1]
                ficherosT += datos[cont][2]
                paginasT += datos[cont][3]
                usuariosT += datos[cont][4]
                usuariosactivosT += datos[cont][5]
                adminsT += datos[cont][6]
                editsT += datos[cont][7]

                cont += 1

        # Ordenar por número de artículos
        datos.sort(key=itemgetter(1), reverse=True)

        # Añadir el total al final de la lista
        datos.append(['total', articulosT, ficherosT, paginasT, usuariosT, usuariosactivosT, adminsT, editsT])

        return datos


def generar_texto(datos, idioma):
        """
        Genera el texto correspondiente para cada idioma.
        """
        new = parametros[idioma]['principio'] + u'<onlyinclude>{{#switch:{{{1}}}\n'
        
        for elem in datos:
                new += u"| " + elem[0] + u' = {{#switch:{{{2}}}'
                
                if elem[0]==idioma and idioma<>None:
                        new += u"\n\t| NUMBEROFARTICLES | ARTICLES = {{NUMBEROFARTICLES:R}}"
                        new += u"\n\t| NUMBEROFFILES | FILES = {{NUMBEROFFILES:R}}"
                        new += u"\n\t| NUMBEROFPAGES |PAGES = {{NUMBEROFPAGES:R}}"
                        new += u"\n\t| NUMBEROFUSERS | USERS = {{NUMBEROFUSERS:R}}"
                        new += u"\n\t| NUMBEROFACTIVEUSERS | ACTIVEUSERS = {{NUMBEROFACTIVEUSERS:R}}"
                        new += u"\n\t| NUMBEROFADMINS | ADMINS = {{NUMBEROFADMINS:R}}"
                        new += u"\n\t| NUMBEROFEDITS | EDITS = {{NUMBEROFEDITS:R}}"
                else:
                        new += u"\n\t| NUMBEROFARTICLES | ARTICLES = " + str(elem[1])
                        new += u"\n\t| NUMBEROFFILES | FILES = " + str(elem[2])
                        new += u"\n\t| NUMBEROFPAGES | PAGES = " + str(elem[3])
                        new += u"\n\t| NUMBEROFUSERS | USERS = " + str(elem[4])
                        new += u"\n\t| NUMBEROFACTIVEUSERS | ACTIVEUSERS = " + str(elem[5])
                        new += u"\n\t| NUMBEROFADMINS | ADMINS = " + str(elem[6])
                        new += u"\n\t| NUMBEROFEDITS | EDITS = " + str(elem[7])

                new += u"\n\t| 0 }}\n"

        new += u"| 0 }}</onlyinclude>" + parametros[idioma]['final']

        return new

def guardar(datos, idioma):
        """
        Guarda el texto generado para el idioma seleccionado.
        """
        page = wikipedia.Page(idioma, parametros[idioma]['pagina'])
        new = generar_texto(datos, idioma)
        page.put(new, parametros[idioma]['resumen'])



datos = generar_datos()
wikipedia.output('\n')

if modo not in parametros:
        for i in parametros:
                guardar(datos, i)
else:
        guardar(datos, modo)
